package org.util; /**
 * 工具类
 * @author yang
 * @date 2023-11-20
 */

import java.sql.*;

/**
 * 连接数据库的工具类 - 文档注释应用在类、方法、字段上面
 * alt + enter
 * ctrl + alt + L

 */
public class DbUtil {
    /**
     * 连接驱动程序 - JDK 中的 JDBC 只是接口，具体实现由数据库厂商实现 - 需要到相关官网下载驱动程序（JDBC实现） - jar包
     */
    private static final String DRIVER = "com.mysql.cj.jdbc.Driver";
    /**
     * 连接URL - jdbc:数据库类型://服务器地址:端口号;DatabaseName=数据库名
     * 其中，服务器地址：还可以是IP地址 或 域名 ; localhost表示本地服务器
     */
    private static final String URL = "jdbc:mysql://127.0.0.1:3306/jdbc_test?serverTimezone=utf8&useSSL=false&serverTimezone=UTC&rewriteBatchedStatements=true";
    /**
     * 帐号
     */
    private static final String USER = "root";
    /**
     * 密码
     */
    private static final String PASS = "ysh@990123";

    static {
        /*
         * 加载驱动程序
         */
        try {
            Class.forName(DRIVER);
        } catch (ClassNotFoundException e) {
            System.out.println("加载驱动程序失败...");
            e.printStackTrace();
        }
    }

    /**
     * 获取连接对象 -- Java程序 与 数据库之间的桥梁
     *
     * @return
     */
    public static Connection getConnection() {
        Connection conn = null;
        try {
            conn = DriverManager.getConnection(URL, USER, PASS);
        } catch (SQLException e) {
            System.out.println("获取连接对象失败...");
            e.printStackTrace();
        }
        return conn;
    }

    /**
     * 关闭相关的 JDBC 对象
     * <p>
     * DriverManager：驱动管理对象，获取连接对象
     * <p>
     * DriverManager.getConnection(URL, USER, PASS);
     * <p>
     * ResultSet：结果集对象 用于接收查询数据时，返回的结果
     * <p>
     * Statement：语句对象 用于执行SQL语句（PreparedStatement、CallableStatement）
     * 增、删、改：executeUpdate() 查询：executeQuery()
     * <p>
     * <p>
     * Connection：连接对象 建立JAVA程序与数据库之间的桥梁
     *
     * @param rst
     * @param stmt 父类对象可以接收子类对象 - 多态
     * @param conn
     */
    private Connection conn;
    private PreparedStatement psmt;
    private ResultSet rs;
    public void closeAll() {
        if (rs != null) {
            try {
                rs.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }

        if (psmt != null) {
            try {
                psmt.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }

        if (conn != null) {
            try {
                conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }



    //添加、查询、修改方法
    public int executeUpdate(String sql,Object...params) throws SQLException {
        conn = getConnection();
        psmt = conn.prepareStatement(sql);
        if (params!=null) {
            for (int i = 0; i < params.length; i++) {
                psmt.setObject(i+1, params[i]);
            }
        }
        int rows = psmt.executeUpdate();
        closeAll();
        return rows;
    }

    //查询方法
    public ResultSet executeQuery(String sql,Object...params) throws SQLException {
        conn = getConnection();
        psmt = conn.prepareStatement(sql);
        if (params!=null) {
            for (int i = 0; i < params.length; i++) {
                psmt.setObject(i+1, params[i]);
            }
        }
        rs = psmt.executeQuery();
        return rs;
    }

    public static void close(Connection conn, Statement stmt, ResultSet rst) {
        if (rst != null) {
            try {
                rst.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }

        if (stmt != null) {
            try {
                stmt.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }

        if (conn != null) {
            try {
                conn.close();
            } catch (SQLException e) {
                e.printStackTrace();
            }
        }
    }

    /**
     * 定义主方法，测试是否连接成功
     *
     * @param args
     */
    public static void main(String[] args) {
        Connection conn = DbUtil.getConnection();
        System.out.println(conn);
//        try {
//            new DbUtil().executeUpdate("delete from staffs where name = '王五'");

//            new DbUtil().executeUpdate("insert into staffs values(default,'王五',20,12323232322,'员工',default,default)");
//            ResultSet rs = new DbUtil().executeQuery("select name from staffs");
//            while(rs.next()){
//                System.out.println(rs.getString(1));
//            }
//        } catch (SQLException e) {
//            throw new RuntimeException(e);
//        }
    }
}
